python convert list comprehension to generator